package timer;

import java.util.concurrent.PriorityBlockingQueue;

/**
 * Created with IntelliJ IDEA.
 * Description:
 * User: 74646
 * Date: 2022-05-21
 * Time: 20:59
 */
//描述任务
class MyTask implements Comparable<MyTask>{
    //任务具体要做什么
    private Runnable runnable;
    //要执行任务的时间（保存任务要执行的毫秒级时间戳）
    private long time;

    //after 是一个时间间隔，不是绝对的时间戳
    public MyTask(Runnable runnable, long delay) {
        this.runnable = runnable;
        // 相当于告诉一个任务，要求在1个小时后执行，不会指定具体的时间去执行
        this.time = System.currentTimeMillis()+delay;
    }
    public void run(){
        runnable.run();
    }
    //获取任务时间
    public long getTime(){
        return time;
    }

    @Override
    public int compareTo(MyTask o) {
        //因为时间设置的是long类型，compareTo方法返回的int类型，所以需要类型转换
        return (int) (this.time-o.time);
    }
}

//组织任务
class MyTimer{
    //定时器内部需要存放多个任务
    private PriorityBlockingQueue<MyTask> queue = new PriorityBlockingQueue<>();

    //实现 schedule 方法,将任务放到队列中
    //将任务注册到 MyTimer 中
    public void schedule(Runnable runnable,long delay){
        MyTask task = new MyTask(runnable,delay);
        queue.put(task);
        //每次新的任务插入成功之后，就唤醒一下扫描线程，让线程重新检查队首任务，看是否到了要执行任务的时间
        synchronized (locker){
            locker.notify();
        }
    }
    private Object locker = new Object();
    public MyTimer(){
        Thread t = new Thread(()->{
            while (true){
//                MyTask task = queue.peek();
//                long curTime = System.currentTimeMillis();
//                if(curTime> task.getTime()){
//                    task.run();
//                }

                try {
                    //获取队首元素
                    MyTask task = queue.take();
                    //获取当前时间
                    long curTime = System.currentTimeMillis();
                    //比较当前时间与任务时间
                    if(curTime<task.getTime()){
                        //时间还没到，将任务放回到队列中
                        queue.put(task);
                        //指定一个等待时间
                        synchronized (locker){
                            locker.wait(task.getTime()-curTime);
                        }
                    }else {
                        //时间到了，执行这个任务
                        task.run();
                    }
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });
        t.start();
    }
}

public class Demo2 {
    public static void main(String[] args) {
        MyTimer timer = new MyTimer();
        timer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("hello timer");
            }
        },3000);
        System.out.println("main");
    }
}
